python实现从文件查找关键字

最新推荐文章于 2024-09-18 16:28:40 发布

qq_1249870753

最新推荐文章于 2024-09-18 16:28:40 发布

阅读量1.2w

点赞数 11

分类专栏： python 人工智能文章标签： python

本文链接：https://blog.csdn.net/fengzongfu/article/details/104830964

版权

人工智能同时被 2 个专栏收录

73 篇文章 0 订阅

订阅专栏

python

25 篇文章 0 订阅

订阅专栏

要求

写一个函数，传入一个路径和一个关键字（关键字是文件内容），找到文件内容里面有这个关键字的文件

代码

import os
def find_file(keywords,path='/Users/wym/Desktop/work/Besttest/笔记/day6'):
    file_list = get_all_file(path)
    for file in file_list:
        if file.endswith('.py') or file.endswith('.txt'):
            with open(file,'r',encoding='utf-8') as fr:
                if keywords in fr.read():
                    print(file)
                    print('=='*20)
def get_all_file(path):
    file_list = []
    for cur_path, cur_dirs, cur_files in os.walk(path):
        for name in cur_files:
            file_list.append(os.path.join(cur_path,name))
    return file_list
keywords = input("Please input the key words that you want to search:")
path = input("Please input the filepath:")
find_file(keywords,path)